# -*- coding: utf-8 -*-
# time: 2025/5/12 08:43
# file: coqui_ts.py
# author: hanson
"""

Coqui TTS（快速轻量）
pip install TTS torchaudio
tts --text "text for TTS" --out_path ./test_speech.wav
"""

from TTS.api import TTS
import torch

# 检查可用设备
device = "cuda" if torch.cuda.is_available() else "cpu"

# 初始化模型（使用小模型 xtts_v1.1）
tts = TTS(model_name="tts_models/multilingual/multi-dataset/xtts_v1.1", progress_bar=False).to(device)

# 文字转语音（支持中文）
text = "欢迎使用文字转语音模型，这是一个Python示例。"
output_path = "./output.wav"

# 合成语音（克隆默认音色，如需自定义音色需提供参考音频）
tts.tts_to_file(
    text=text,
    speaker_wav="reference.wav",  # 替换为你的参考音频文件（如无则用默认音色）
    file_path=output_path,
    language="zh-cn"  # 语言代码
)

print(f"语音已生成到: {output_path}")